package JZ.JZ29;

/**
 * @author corn
 * @create 2020-08-09-19:38
 */

import java.util.ArrayList;

/**
 * JZ29 最小的K个数
 * 题目描述
 * 输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。
 */
public class Solution {
    //其他人通过代码：运行时间: 13ms 占用内存: 9448KB
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();

        if(input.length < k || k == 0)
            return list;
        //将前k个数添加到list中
        for (int i = 0; i < k; i++)
            list.add(input[i]);
        //遍历后续的数，与list中最大的数小的的数，将这个两个数进行交换
        //由于list存储的时有序的、可重复的数据，所以只要确保list中的数是最小的k个数就行
        for (int i = k; i < input.length; i++) {
            int j = this.getMax(list);
            int temp = (Integer) list.get(j);
            if (input[i] < temp)
                list.set(j, input[i]);

        }
        return list;
    }
    //找到list最大值的下标
    public int getMax(ArrayList<Integer> list) {
        int max = list.get(0);
        int j = 0;
        for (int i = 0; i < list.size(); i++) {
            if (list.get(i) > max) {
                max = list.get(i);
                j = i;
            }
        }
        return j;
    }
    /*//自己编写：运行时间: 14ms 占用内存: 9668KB
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> integers = new ArrayList<>();
        if(input == null ||input.length == 0 || input.length < k){
            return integers;
        }
        for (int i = 0; i < k; i++) {
            for (int j = input.length - 1; j >0 ; j--) {
                if(input[j] < input[j-1]){
                    int temp = input[j-1];
                    input[j-1] = input[j];
                    input[j] = temp;
                }
            }
            integers.add(input[i]);
        }
        return integers;
    }*/
}
